from setuptools import setup, find_packages

requirements_file = './requirements.txt'
version_file = './READ_pytorch/version.py'

def read_version_file():
    with open(version_file, 'r') as f:
        exec(compile(f.read(), version_file, 'exec'))
    return locals()

def read_requirements_file():
    with open(requirements_file, 'r') as f:
        requirements = f.readlines()
    requirements = [r.strip() for r in requirements]
    return requirements

def readme():
    with open('README.md', encoding='utf-8') as f:
        content = f.read()
    return content

if __name__ == '__main__':
    info = read_version_file()
    requirements = read_requirements_file()

    setup(
        name='READ_pytorch',
        version=info['__version__'],
        description='Unsupervised Anomaly Localization Toolbox and Benchmark',
        url='https://git.openi.org.cn/OpenI/READ_pytorch',
        license='Apache License 2.0',
        author='TCL MV Lab',
        author_email='chao46.zhang@tcl.com',
        python_requires='>=3, <4',
        long_description=readme(),
        long_description_content_type='text/markdown',
        install_requires=requirements,
        packages=find_packages(),
        classifiers=[
            # How mature is this project? Common values are
            #   3 - Alpha
            #   4 - Beta
            #   5 - Production/Stable
            'Development Status :: 3 - Alpha',
            'License :: OSI Approved :: Apache Software License',
            'Operating System :: OS Independent',
            'Programming Language :: Python :: 3',
            'Programming Language :: Python :: 3.6',
            'Programming Language :: Python :: 3.7',
            'Programming Language :: Python :: 3.8',
        ],
    )